package com.hevery.security.core.validate.code;

import com.hevery.security.core.validate.code.image.ImageCode;
import com.hevery.security.core.validate.code.sms.SmsCodeSender;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.social.connect.web.HttpSessionSessionStrategy;
import org.springframework.social.connect.web.SessionStrategy;
import org.springframework.web.bind.ServletRequestBindingException;
import org.springframework.web.bind.ServletRequestUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.ServletWebRequest;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Map;

/**
 * @author: zhy
 * @description:
 * @date: created on 2017/10/26
 * @modified by:
 */
@RestController
public class ValidateCodeController {

//    public static final String SESSION_KEY = "SESSION_KEY_IMAGE_CODE";
//
//    /**Spring中操作Session的工具类*/
//    private SessionStrategy sessionStrategy = new HttpSessionSessionStrategy();
//
//    /**此处注入一个ValidateCodeGenerator接口的实现类ImageCodeGenerator，图片验证码生成器*/
//    @Autowired
//    private ValidateCodeGenerator imageCodeGenerator;
//
//    /**此处注入一个ValidateCodeGenerator接口的实现类SmscodeGenerator，短信验证码生成器*/
//    @Autowired
//    private ValidateCodeGenerator smsCodeGenerator;
//
//    @Autowired
//    private SmsCodeSender smsCodeSender;
//
//    @GetMapping("/code/image")
//    public void createCode(HttpServletRequest request, HttpServletResponse response) throws IOException{
//
//        //生成ImageCode(封装了图形验证码)------>第一步：生成验证码
//        ImageCode imageCode = (ImageCode) imageCodeGenerator.generate(new ServletWebRequest(request));
//
//        //将随机数存到Session中，第一个参数用于把请求传进去，第二个参数是Session的key，第三个是Session的value。
//        //------>第二步：将生成的验证码放在Session中
//        sessionStrategy.setAttribute(new ServletWebRequest(request), SESSION_KEY, imageCode);
//
//        //将生成的图片写到接口的响应中----->将生成的验证码发送出去
//        ImageIO.write(imageCode.getImage(), "JPEG", response.getOutputStream());
//    }
//
//    @GetMapping("/code/sms")
//    public void createSmsCode(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletRequestBindingException {
//
//        //生成smsCode(封装了短信验证码)----->第一步：生成验证码
//        ValidateCode smsCode = smsCodeGenerator.generate(new ServletWebRequest(request));
//
//        //将随机数存到Session中，第一个参数用于把请求传进去，第二个参数是Session的key，第三个是Session的value。
//        //------>第二步：将生成的验证码放在Session中
//        sessionStrategy.setAttribute(new ServletWebRequest(request), SESSION_KEY, smsCode);
//
//        //获取请求中传入的手机号，getRequiredStringParameter是值请求中必须包含这个各参数
//        String mobile = ServletRequestUtils.getRequiredStringParameter(request, "mobile");
//
//        //通过短信服务商把短信验证码发送出去------>第三步：将生成的验证码发送出去
//        smsCodeSender.send(mobile, smsCode.getCode());
//    }

    @Autowired
    private Map<String, ValidateCodeProcessor> validateCodeProcessors;


    /**
     * 把之前的连个服务合并成一个服务，使用@PathVariable注解从请求中获取要响应的类型。
     * 仍然使用依赖搜索的原理，在validateCodeProcessors这个map中查找对应的ValidateCodeProcessor的实现，找到后调用其create方法。
     * @param request
     * @param response
     * @param type
     * @throws Exception
     */
    @GetMapping("/code/{type}")
    public void createCode(HttpServletRequest request, HttpServletResponse response, @PathVariable String type) throws Exception{
        validateCodeProcessors.get(type + "CodeProcessor").create(new ServletWebRequest(request, response));
    }

}
